how to draw cool shapes in python